/**
 * Software License, Version 1.0
 * 
 * Copyright 2003 The Trustees of Indiana University.  All rights reserved.
 * 
 *
 *Redistribution and use in source and binary forms, with or without 
 *modification, are permitted provided that the following conditions are met:
 *
 *1) All redistributions of source code must retain the above copyright notice,
 * the list of authors in the original source code, this list of conditions and
 * the disclaimer listed in this license;
 *2) All redistributions in binary form must reproduce the above copyright 
 * notice, this list of conditions and the disclaimer listed in this license in
 * the documentation and/or other materials provided with the distribution;
 *3) Any documentation included with all redistributions must include the 
 * following acknowledgement:
 *
 *"This product includes software developed by the Community Grids Lab. For 
 * further information contact the Community Grids Lab at 
 * http://communitygrids.iu.edu/."
 *
 * Alternatively, this acknowledgement may appear in the software itself, and 
 * wherever such third-party acknowledgments normally appear.
 * 
 *4) The name Indiana University or Community Grids Lab or NaradaBrokering, 
 * shall not be used to endorse or promote products derived from this software 
 * without prior written permission from Indiana University.  For written 
 * permission, please contact the Advanced Research and Technology Institute 
 * ("ARTI") at 351 West 10th Street, Indianapolis, Indiana 46202.
 *5) Products derived from this software may not be called NaradaBrokering, 
 * nor may Indiana University or Community Grids Lab or NaradaBrokering appear
 * in their name, without prior written permission of ARTI.
 * 
 *
 * Indiana University provides no reassurances that the source code provided 
 * does not infringe the patent or any other intellectual property rights of 
 * any other entity.  Indiana University disclaims any liability to any 
 * recipient for claims brought by any other entity based on infringement of 
 * intellectual property rights or otherwise.  
 *
 *LICENSEE UNDERSTANDS THAT SOFTWARE IS PROVIDED "AS IS" FOR WHICH NO 
 *WARRANTIES AS TO CAPABILITIES OR ACCURACY ARE MADE. INDIANA UNIVERSITY GIVES
 *NO WARRANTIES AND MAKES NO REPRESENTATION THAT SOFTWARE IS FREE OF 
 *INFRINGEMENT OF THIRD PARTY PATENT, COPYRIGHT, OR OTHER PROPRIETARY RIGHTS. 
 *INDIANA UNIVERSITY MAKES NO WARRANTIES THAT SOFTWARE IS FREE FROM "BUGS", 
 *"VIRUSES", "TROJAN HORSES", "TRAP DOORS", "WORMS", OR OTHER HARMFUL CODE.  
 *LICENSEE ASSUMES THE ENTIRE RISK AS TO THE PERFORMANCE OF SOFTWARE AND/OR 
 *ASSOCIATED MATERIALS, AND TO THE PERFORMANCE AND VALIDITY OF INFORMATION 
 *GENERATED USING SOFTWARE.
 */
package cgl.narada.benchmark.security;

import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.Random;

import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;

import cgl.narada.service.ServiceException;
import cgl.narada.service.security.impl.EntityOperationsImpl;
import cgl.narada.service.timer.ClockI;
import cgl.narada.service.timer.HRClock;
import cgl.narada.util.SystemInit;
import cgl.narada.util.SystemNotInitializedException;

/**
 * Benchmarks Encryption and Decryption timings <br>
 * Created on Apr 19, 2006
 * 
 * @author Harshawardhan Gadgil (hgadgil@grids.ucs.indiana.edu)
 */
public class EncryptionDecryption {
    public static void main(String[] args) {

        SystemInit.init();

        String nbHome = null;
        try {
            nbHome = SystemInit.getInitData(SystemInit.SYS_NB_HOME);
        } catch (SystemNotInitializedException e3) {
            // TODO Auto-generated catch block
            e3.printStackTrace();
        }

        ClockI hrclock = HRClock.getClock(nbHome);
        EntityOperationsImpl eoi = null;
        try {
            eoi = new EntityOperationsImpl();
        } catch (ServiceException e3) {
            // TODO Auto-generated catch block
            e3.printStackTrace();
        }

        KeyGenerator kg = null;
        try {
            kg = KeyGenerator.getInstance("AES");
        } catch (NoSuchAlgorithmException e2) {
            e2.printStackTrace();
        }

        kg.init(256, new SecureRandom());
        SecretKey secretKey = kg.generateKey();

        int size = 16 * 1024;

        byte[] originalPayload = new byte[size];
        Random r = new Random();

        FileOutputStream fout = null;
        try {
            fout = new FileOutputStream(
                    "/home/hgadgil/EncryptionDecryption.dat");
        } catch (FileNotFoundException e1) {
            // TODO Auto-generated catch block
            e1.printStackTrace();
        }

        // Set the default properties...
        String mode = "CBC", padding = "PKCS7Padding", signingAlgorithm = "SHA1withRSA";
        String algModePadding = secretKey.getAlgorithm() + "/" + mode + "/"
                + padding;

        long t1;
        long start;

        // Step 1: Encrypt payload

        byte[] encryptedPayload = null;

//        long[] v1 = new long[101];

        for (int i = 0; i <= 100; i++) {
            System.out.println("--> " + i);
            r.nextBytes(originalPayload);

            start = hrclock.getTimeMicroseconds();

            try {
                byte[] iv = eoi.getIV(secretKey.getAlgorithm(), mode, padding);

                encryptedPayload = eoi.encryptPayload(secretKey,
                        originalPayload, algModePadding, iv, "BC");
            } catch (ServiceException e) {
                e.printStackTrace();
            }

            t1 = hrclock.getTimeMicroseconds();

//            v1[i] = t1 - start;

            try {
                fout.write(("" + (t1 - start) + "\n").getBytes());
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        // Step 2: Decrypt payload

//        long[] v2 = new long[101];

        for (int i = 0; i <= 100; i++) {
            System.out.println("--> " + i);
            r.nextBytes(originalPayload);

            start = hrclock.getTimeMicroseconds();

            try {
                byte[] iv = eoi.getIV(secretKey.getAlgorithm(), mode, padding);

                byte[] decryptedPayload = eoi.decryptPayload(secretKey,
                        encryptedPayload, algModePadding, iv, "BC");
            } catch (ServiceException e) {
                e.printStackTrace();
            }

            t1 = hrclock.getTimeMicroseconds();

//            v2[i] = t1 - start;

            try {
                fout.write(("" + (t1 - start) + "\n").getBytes());
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        try {
            fout.close();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        // Analyze:

//        long[] v = new long[100];

    }

    // private static void analyze(long[] data, int skipFirst) {
    // int i = data.length - skipFirst;
    //
    // long[] v = new long[i];
    // int c = 0;
    // for (int x = i; x < data.length; x++)
    // v[c++] = data[x];
    //
    // System.out.println("MEAN: " + StatUtils.computeMeanWithoutOutliers(v,
    // 100));
    // System.out.println("MEAN: " +
    // StatUtils.computeStandardDeviationWithoutOutliers(v, 100));
    // System.out.println("MEAN: " + StatUtils.computeStandardErrorOfMean(v,
    // 100));
    //
    // }

}
